How to use keras in neural network